Amazon Data Science Interview: Linear Regression